package cn.tedu.weibo.controller;

import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.io.IOException;
import java.util.UUID;

@RestController
public class UploadController {
    @RequestMapping("/upload")
    public String upload(MultipartFile pic) throws IOException {
        //MultipartFile 类型为pic
        //pic 代表客户端上传的文件参数
        System.out.println("pic = " + pic);
        //得到图片的原始文件名  a.jpg
        String filename= pic.getOriginalFilename();
        System.out.println("原始文件名:" + filename);
        //得到原始文件名的后缀  .jpg
        String suffix=filename.substring(filename.lastIndexOf("."));
        //得到唯一文件名 UUID.randomUUID()获取唯一标识符
        filename= UUID.randomUUID()+suffix;
        System.out.println("唯一文件名:"+filename);
        //准备保存图片的路径
        String dirPath="d:/files";
        File dirFile=new File(dirPath);
        //判断如果不存在  则创建文件夹
        if(!dirFile.exists()){
            dirFile.mkdirs();//创建文件夹
        }
        //得到文件的完整路径  d:files/xxxx.jpg
        String filePath=dirPath+"/"+filename;
        //把图片保存到上面的路径中  抛出异常
        pic.transferTo(new File(filePath));

        //把图片的url路径响应给客服端  /文件名
        return "/" +filename;
    }
    @RequestMapping("/remove")
    public void remove(String url){
        //删除磁盘上的文件
        new File("d:/files"+url).delete();
    }
}
